package com.sl.common.util;/*
 * Copyright [2022] [https://www.xiaonuo.vip]
 *
 * Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
 *
 * 1.请不要删除和修改根目录下的LICENSE文件。
 * 2.请不要删除和修改Snowy源码头部的版权声明。
 * 3.本项目代码可免费商业使用，商业使用请保留源码和相关描述文件的项目出处，作者声明等。
 * 4.分发源码时候，请注明软件出处 https://www.xiaonuo.vip
 * 5.不可二次分发开源参与同类竞品，如有想法可联系团队xiaonuobase@qq.com商议合作。
 * 6.若您的项目无法满足以上几点，需要更多功能代码，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip
 */

import cn.hutool.core.io.FileUtil;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;

import java.io.File;
import java.io.IOException;
import java.io.OutputStream;

/**
 * 文件下载工具类，使用本类前，对参数校验的异常使用CommonResponseUtil.renderError()方法进行渲染
 *
 * @author xuyuxiang
 * @date 2020/8/5 21:45
 */
@Slf4j
public class CommonDownloadUtil {

    /**
     * 下载文件
     *
     * @param file     要下载的文件
     * @param response 响应
     * @author xuyuxiang
     * @date 2020/8/5 21:46
     */
    public static void download(File file, HttpServletResponse response) {
        download(file.getName(), FileUtil.readBytes(file), response);
    }

    /**
     * 下载文件
     *
     * @author xuyuxiang
     * @date 2022/7/31 10:57
     */
    public static void download(String fileName, byte[] fileBytes, HttpServletResponse response) {
        try {
//            response.setHeader("Content-Disposition", "attachment;filename=\"" + URLUtil.encode(fileName)+ "\"");
            response.addHeader("Content-Length", "" + fileBytes.length);
//            response.setHeader("Access-Control-Allow-Origin", "*");
////            response.setHeader("Access-Control-Expose-Headers", "Content-Disposition");
            String suffix = FileUtil.getSuffix(fileName);
            response.setContentType("image/"+suffix);
//            response.setHeader("Cache-Control", "no-cache");
//            response.setHeader("Accept-Ranges", "bytes");
//            response.setHeader("expires", "Wed, 07 May 2025 00:28:25 GMT");
//            response.setHeader("age", "1850351");
//            response.setHeader("date", "Mon, 28 Apr 2025 10:27:59 GMT");
//            response.setHeader("etag", "c927482db1b358521cec9de04627a562");
//            response.setHeader("Last-modified", "Sat, 10 Jan 1970 00:00:00 GMT");

            //ByteArrayInputStream inputStream = new ByteArrayInputStream(fileBytes);
            // 循环，边读取边输出，可避免大文件时OOM
            try (OutputStream os = response.getOutputStream()) {
                os.write(fileBytes);
//                byte[] bytes = new byte[1024];
//                int readLength;
//                while ((readLength = inputStream.read(bytes)) != -1) {
//                    os.write(bytes, 0, readLength);
//                }
            }
        } catch (IOException e) {
            log.error(">>> 文件下载异常：", e);
        }
    }
}
